Fill the Diagonal of Each Matrix in a 3D numpy Array with a Vector